
#include "register/tilingdata_base.h"

namespace optiling {
BEGIN_TILING_DATA_DEF(BallQueryTilingData)
  TILING_DATA_FIELD_DEF(uint32_t, B);
  TILING_DATA_FIELD_DEF(uint32_t, N);
  TILING_DATA_FIELD_DEF(uint32_t, M);
  TILING_DATA_FIELD_DEF(uint32_t, NAlign);  
  TILING_DATA_FIELD_DEF(uint32_t, MAlign);
  TILING_DATA_FIELD_DEF(uint32_t, baseB);
  TILING_DATA_FIELD_DEF(float, neg_min);
  TILING_DATA_FIELD_DEF(float, neg_max);
  TILING_DATA_FIELD_DEF(int32_t, sample_num);
  TILING_DATA_FIELD_DEF(int32_t, sample_numAlign);
  TILING_DATA_FIELD_DEF(int32_t, batch_shape);
END_TILING_DATA_DEF;

REGISTER_TILING_DATA_CLASS(BallQuery, BallQueryTilingData)
}
